package com.brainhealth.customer.service;

import com.alibaba.fastjson.JSONObject;
import com.brainhealth.common.enums.TokenTypeEnum;

/**
 * 文心一言服务
 * @author wll
 */
public interface IErnieBotService {

    /**
     * 获取 accessToken
     * @return
     */
    String getAccessToken();

    /**
     * 获取答案
     * @param object
     * @return
     */
    String getErnieBotResult(JSONObject object);

    /**
     * 获取总字数
     * @param object
     * @return
     */
    Long getTotalTokens(JSONObject object, TokenTypeEnum tokenTypeEnum);

    /**
     * 获取答案
     * @param accessToken access_tokens
     * @param prompt 问题
     * @return
     */
    JSONObject getCompletion(String accessToken,String prompt);

    /**
     * 刷新token
     * @return
     */
    Integer refreshToken();
}
